from WRASimulationService import wRASimulationService
from District import district
from TimeBean import timeBean
from Intake import intake
from River import river
from Station import station
district1 = district('01-010100','北京潮白河山区',4725.421895
                     ,[88.91001888,89.08804018,89.81307766,90.39916878,90.49264875,90.76839557,92.46860365,94.70934383,96.29508381,98.10665027,101.6245176]
                     ,[52.60770328,52.67935485,53.30655068,53.71239784,53.03985903,53.5692132,56.1180823,58.71251004,60.96442815,63.35772406,70.19847002]
                     ,[36.30230832,36.40868532,36.50662687,36.68677094,37.45268983,37.19918237,36.35052172,35.9968338,35.33056303,34.74892622,31.42585473]
                     ,[48604.7717],[1046.843754,1118.946539,1192.085946,1221.86002,1225.257835]
                     ,[0.71, 0.75, 0.82],18695.914254578216,[1.382827501987302,
                        1.8441201022909388,
                        7.661477155058188,
                        1508.9315715617156,
                        4213.71025352189,
                        3942.1459429748224,
                        3759.325570060147,
                        3477.7108840248134,
                        443.87751379892853,
                        20.159870423455693,
                        1.1003033095307595,
                        0.0
                    ],7293.607308460832, [
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027,
                        607.8006090384027
                    ],29356.851098778134,[
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511,
                        2446.404258231511
                    ],940.9067976+1447.3435082730002,[120.66601132,120.68188595, 120.92362614, 129.53215708 ,147.49026187,
 147.672068 ,  145.30876735, 144.98791373, 124.25746497 ,121.42704316,
 120.71764778, 120.61195902],200,200,w_wds=10000,w_other=0)
timeBean1 = timeBean(75368,0,0,0,0,{},0,0,0,0)
timebeans = [timeBean1 for i in range(12)]
districts=[district1]
station1 = station('10','水库1',12,100,500,100,30,10,district1,300,timebeans,100,100,100,100,100,100,districts)
intake1 = intake('01','取水口1',12,100,500,100,30,10,district1,300,timebeans)
intakes=[intake1]
stations=[station1]
elements1=[station1,intake1]
river1=river('001','支流1',timebeans,True,False,[],stations,intakes,elements1,districts)
rivers=[river1]
elements = [river1,station1,intake1]
river=river('000','幹流',timebeans,False,True,rivers,stations,intakes,elements,districts)
w = wRASimulationService()
w.execute(river,12,0,['外调水','水库水','本地水','地下水'],[2,1,0,0])